package com.springboot.util.file;


import lombok.extern.slf4j.Slf4j;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;


/**
 * @description: //创建新文件和目录
 * @author: zyj
 * @create: 2020-09-09 18:11
 **/
@Slf4j
public class MyFileUtil {

    // 验证字符串是否为正确路径名的正则表达式
    private static String matches = "[A-Za-z]:\\\\[^:?\"><*]*";
    // 通过 sPath.matches(matches) 方法的返回值判断是否正确
    // sPath 为路径字符串
    static boolean flag = false;
    //static File file;

    /**
    * @Description: 根据路径删除指定的目录或文件，无论存在与否
    * @Param: [deletePath]
    * @return: boolean
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static boolean DeleteFolder(String deletePath) {
        flag = false;
        if (deletePath.matches(matches)) {
            File file = new File(deletePath);
            if (!file.exists()) {// 判断目录或文件是否存在
                return flag; // 不存在返回 false
            } else {

                if (file.isFile()) {// 判断是否为文件
                    return deleteFile(deletePath);// 为文件时调用删除文件方法
                } else {
                    return deleteDirectory(deletePath);// 为目录时调用删除目录方法
                }
            }
        } else {
            log.info("要传入正确路径！");
            return false;
        }
    }

    /**
    * @Description: 删除单个文件
    * @Param: [filePath]
    * @return: boolean
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static boolean deleteFile(String filePath) {
        flag = false;
        File file = new File(filePath);
        if (file.isFile() && file.exists()) {// 路径为文件且不为空则进行删除
            file.delete();// 文件删除
            flag = true;
        }
        return flag;
    }


    /**
    * @Description: 删除目录（文件夹）以及目录下的文件
    * @Param: [dirPath]
    * @return: boolean
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static boolean deleteDirectory(String dirPath) {//
        // 如果sPath不以文件分隔符结尾，自动添加文件分隔符
        if (!dirPath.endsWith(File.separator)) {
            dirPath = dirPath + File.separator;
        }
        File dirFile = new File(dirPath);
        // 如果dir对应的文件不存在，或者不是一个目录，则退出
        if (!dirFile.exists() || !dirFile.isDirectory()) {
            return false;
        }
        flag = true;
        File[] files = dirFile.listFiles();// 获得传入路径下的所有文件
        for (int i = 0; i < files.length; i++) {// 循环遍历删除文件夹下的所有文件(包括子目录)
            if (files[i].isFile()) {// 删除子文件
                flag = deleteFile(files[i].getAbsolutePath());
                log.info(files[i].getAbsolutePath() + " 删除成功");
                if (!flag) {
                    break;// 如果删除失败，则跳出
                }
            } else {// 运用递归，删除子目录
                flag = deleteDirectory(files[i].getAbsolutePath());
                if (!flag) {
                    break;// 如果删除失败，则跳出
                }
            }
        }
        if (!flag) {
            return false;
        }
        if (dirFile.delete()) {// 删除当前目录
            return true;
        } else {
            return false;
        }
    }

    /**
     * @Description:  创建单个文件
     * @Param: [filePath]
     * @return: boolean
     * @Author: zyj
     * @Date: 2020/9/9
     */
    public static File createFile(byte[] bytes,String filePath,String fileName) {
        File file = new File(filePath+File.separator+fileName);
        if (file.exists()) {// 判断文件是否存在
            log.info("目标文件已存在" + filePath);
            return null;
        }
        if (filePath.endsWith(File.separator)) {// 判断文件是否为目录
            log.info("目标文件不能为目录！");
            return null;
        }
        if (!file.getParentFile().exists()) {// 判断目标文件所在的目录是否存在
            // 如果目标文件所在的文件夹不存在，则创建父文件夹
            log.info("目标文件所在目录不存在，准备创建它！");
            if (!file.getParentFile().mkdirs()) {// 判断创建目录是否成功
                log.error("创建目标文件所在的目录失败！");
                return null;
            }
        }

        BufferedOutputStream bos = null;
        FileOutputStream fos = null;
        try {
            file = new File(filePath + File.separator + fileName);
            fos = new FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bytes);
            return file;
        } catch (IOException e) {// 捕获异常
            e.printStackTrace();
            log.error("创建文件失败！" + e.getMessage());
            return null;
        }finally {
            if(null != bos){
                try {
                    bos.close();
                } catch (IOException e) {
                    log.error("创建文件bos关闭失败！", e);
                }
            }
            if(null != fos){
                try {
                    fos.close();
                } catch (IOException e) {
                    log.error("创建文件fos关闭失败！", e);
                }
            }
        }
    }

    /**
    * @Description:  创建单个文件
    * @Param: [filePath]
    * @return: boolean
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static boolean createFile(String filePath) {
        File file = new File(filePath);
        if (file.exists()) {// 判断文件是否存在
            log.info("目标文件已存在" + filePath);
            return false;
        }
        if (filePath.endsWith(File.separator)) {// 判断文件是否为目录
            log.info("目标文件不能为目录！");
            return false;
        }
        if (!file.getParentFile().exists()) {// 判断目标文件所在的目录是否存在
            // 如果目标文件所在的文件夹不存在，则创建父文件夹
            log.info("目标文件所在目录不存在，准备创建它！");
            if (!file.getParentFile().mkdirs()) {// 判断创建目录是否成功
                log.error("创建目标文件所在的目录失败！");
                return false;
            }
        }
        try {
            if (file.createNewFile()) {// 创建目标文件
                log.info("创建文件成功:" + filePath);
                return true;
            } else {
                log.error("创建文件失败！");
                return false;
            }
        } catch (IOException e) {// 捕获异常
            e.printStackTrace();
            log.error("创建文件失败！" + e.getMessage());
            return false;
        }
    }


    /**
    * @Description:  创建目录
    * @Param: [destDirName]
    * @return: boolean
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static boolean createDir(String destDirName) {
        File dir = new File(destDirName);
        if (dir.exists()) {// 判断目录是否存在
            log.info("目标目录已存在！");
            return true;
        }
        if (!destDirName.endsWith(File.separator)) {// 结尾是否以"/"结束
            destDirName = destDirName + File.separator;
        }
        if (dir.mkdirs()) {// 创建目标目录
            log.info("创建目录成功！" + destDirName);
            return true;
        } else {
            log.error("创建目录失败！");
            return false;
        }
    }


    /**
    * @Description: 创建临时文件
    * @Param: [prefix, suffix, dirName]
    * @return: java.lang.String
    * @Author: zyj
    * @Date: 2020/9/9
    */
    public static String createTempFile(String prefix, String suffix,
                                        String dirName) {
        File tempFile = null;
        if (dirName == null) {// 目录如果为空
            try {
                tempFile = File.createTempFile(prefix, suffix);// 在默认文件夹下创建临时文件
                return tempFile.getCanonicalPath();// 返回临时文件的路径
            } catch (IOException e) {// 捕获异常
                log.error("创建临时文件失败：",  e);
                return null;
            }
        } else {
            // 指定目录存在
            File dir = new File(dirName);// 创建目录
            if (!dir.exists()) {
                // 如果目录不存在则创建目录
                if (MyFileUtil.createDir(dirName)) {
                    log.error("创建临时文件失败，不能创建临时文件所在的目录！");
                    return null;
                }
            }
            try {
                tempFile = File.createTempFile(prefix, suffix, dir);// 在指定目录下创建临时文件
                return tempFile.getCanonicalPath();// 返回临时文件的路径
            } catch (IOException e) {// 捕获异常
                log.error("创建临时文件失败!", e);
                return null;
            }
        }
    }

    /**
     * the traditional io way
     *
     * @param filename
     * @return
     * @throws IOException
     */
    public static byte[] toByteArray(String filename) throws IOException {

        File f = new File(filename);
        if (!f.exists()) {
            throw new FileNotFoundException(filename);
        }

        ByteArrayOutputStream bos = new ByteArrayOutputStream((int) f.length());
        BufferedInputStream in = null;
        try {
            in = new BufferedInputStream(new FileInputStream(f));
            int buf_size = 1024;
            byte[] buffer = new byte[buf_size];
            int len = 0;
            while (-1 != (len = in.read(buffer, 0, buf_size))) {
                bos.write(buffer, 0, len);
            }
            return bos.toByteArray();
        } catch (IOException e) {
            e.printStackTrace();
            throw e;
        } finally {
            try {
                in.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            bos.close();
        }
    }

    /**
     * bfile 需要转换成文件的byte数组
     * filePath  生成的文件保存路径
     * fileName  生成文件后保存的名称如test.pdf，test.jpg等
     */
    public static void getFile(byte[] bfile, String filePath,String fileName) {
        BufferedOutputStream bos = null;
        FileOutputStream fos = null;
        File file = null;
        try {
            File dir = new File(filePath);
            boolean isDir = dir.isDirectory();
            if (!isDir) {// 目录不存在则先建目录
                try {
                    dir.mkdirs();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
            file = new File(filePath + File.separator + fileName);
            fos = new FileOutputStream(file);
            bos = new BufferedOutputStream(fos);
            bos.write(bfile);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException e1) {
                    e1.printStackTrace();
                }
            }
            if (fos != null) {
                try {
                    fos.close();
                } catch (IOException e1) {
                    e1.printStackTrace();
                }
            }
        }
    }


    /**
     * BufferedImage转byte[]
     *
     * @param bImage BufferedImage对象 fileType图片类型
     * @return byte[]
     * @auth zyj
     */
    public static byte[] imageToBytes(BufferedImage bImage, String fileType) throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        try {
            ImageIO.write(bImage, fileType, out);
        } catch (IOException e) {
            log.error("BufferedImage转byte[]失败", e);
            throw e;
        }
        return out.toByteArray();
    }



//    public static void main(String[] args) {
//        String dirName = "E:/createFile/";// 创建目录
//        MyFileUtil.createDir(dirName);// 调用方法创建目录
//        String fileName = dirName + "/file1.txt";// 创建文件
//        MyFileUtil.createFile(fileName);// 调用方法创建文件
//        String prefix = "temp";// 创建临时文件
//        String surfix = ".txt";// 后缀
//        for (int i = 0; i < 10; i++) {// 循环创建多个文件
//            System.out.println("创建临时文件: "// 调用方法创建临时文件
//                    + MyFileUtil.createTempFile(prefix, surfix,
//                    dirName));
//        }
//
//    }

}


